<?php

// +----------------------------------------------------------------------
// | ThinkPHP [ WE CAN DO IT JUST THINK IT ]
// +----------------------------------------------------------------------
// | Copyright (c) 2009 http://thinkphp.cn All rights reserved.
// +----------------------------------------------------------------------
// | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
// +----------------------------------------------------------------------
// | Author: liu21st <liu21st@gmail.com>
// +----------------------------------------------------------------------
// $Id: UploadFile.class.php 2601 2012-01-15 04:59:14Z liu21st $

/**
  +------------------------------------------------------------------------------
 * 文件上传类
  +------------------------------------------------------------------------------
 * @category   ORG
 * @package  ORG
 * @subpackage  Net
 * @author    liu21st <liu21st@gmail.com>
 * @version   $Id: UploadFile.class.php 2601 2012-01-15 04:59:14Z liu21st $
  +------------------------------------------------------------------------------
 */
class UploadFile {//类定义开始
    // 上传文件的最大值

    public $maxSize = -1;
    // 是否支持多文件上传
    public $supportMulti = true;
    // 允许上传的文件后缀
    //  留空不作后缀检查
    public $allowExts = array();
    // 允许上传的文件类型
    // 留空不做检查
    public $allowTypes = array();
    // 使用对上传图片进行缩略图处理
    public $thumb = false;
    // 图库类包路径
    public $imageClassPath = 'Util.Image';
    // 缩略图最大宽度
    public $thumbMaxWidth;
    // 缩略图最大高度
    public $thumbMaxHeight;
    // 缩略图前缀
    public $thumbPrefix = 'thumb_';
    public $thumbSuffix = '';
    // 缩略图保存路径
    public $thumbPath = '';
    // 缩略图文件名
    public $thumbFile = '';
    // 是否移除原图
    public $thumbRemoveOrigin = true;
    // 压缩图片文件上传
    public $zipImages = false;
    // 启用子目录保存文件
    public $autoSub = true;
    // 子目录创建方式 可以使用hash date
    public $subType = 'date';
    public $dateFormat = '';
    public $hashLevel = 3; // hash的目录层次
    // 上传文件保存路径
    public $savePath = "";
    public $autoCheck = true; // 是否自动检查附件
    // 存在同名是否覆盖
    public $uploadReplace = false;
    // 上传文件命名规则
    // 例如可以是 time uniqid com_create_guid 等
    // 必须是一个无需任何参数的函数名 可以使用自定义函数
    public $saveRule = 'uniqid';
    // 上传文件Hash规则函数名
    // 例如可以是 md5_file sha1_file 等
    public $hashType = 'md5';
    // 错误信息
    private $error = '';
    // 上传成功的文件信息
    private $uploadFileInfo;
    //网站配置信息
    public $Config;
    //FTP上传
    public $Ftpstatus = 0;
    public $Ftp;

    /**
      +----------------------------------------------------------
     * 架构函数
      +----------------------------------------------------------
     * @access public
      +----------------------------------------------------------
     */
    public function __construct($maxSize = '', $allowExts = '', $allowTypes = '', $savePath = '', $saveRule = '') {

        if (defined("CONFIG_FTPSTATUS") && CONFIG_FTPSTATUS) {
            import('Ftp');
            $this->Ftp = new Ftp();
            $this->Ftpstatus = 1;
        }
        if (!empty($maxSize) && is_numeric($maxSize)) {
            $this->maxSize = $maxSize;
        }
        if (!empty($allowExts)) {
            if (is_array($allowExts)) {
                $this->allowExts = array_map('strtolower', $allowExts);
            } else {
                $this->allowExts = explode(',', strtolower($allowExts));
            }
        }
        if (!empty($allowTypes)) {
            if (is_array($allowTypes)) {
                $this->allowTypes = array_map('strtolower', $allowTypes);
            } else {
                $this->allowTypes = explode(',', strtolower($allowTypes));
            }
        }
        if (!empty($saveRule)) {
            $this->saveRule = $saveRule;
        } else {
            $this->saveRule = C('UPLOAD_FILE_RULE');
        }

        if ($this->Ftpstatus) {
            $ftpsta = $this->Ftp->connect(CONFIG_FTPHOST, CONFIG_FTPUSER, CONFIG_FTPPASSWORD, CONFIG_FTPPORT, CONFIG_FTPPASV, CONFIG_FTPSSL, CONFIG_FTPTIMEOUT);
            if ($ftpsta == false) {
                $this->error = 'FTP链接失败' . CONFIG_FTPHOST;
            }
        }
    }

    /**
      +----------------------------------------------------------
     * 上传一个文件
      +----------------------------------------------------------
     * @access public
      +----------------------------------------------------------
     * @param mixed $name 数据
     * @param string $value  数据表名
      +----------------------------------------------------------
     * @return string
      +----------------------------------------------------------
     * @throws ThinkExecption
      +----------------------------------------------------------
     */
    private function save($file) {
        $filename = $file['savepath'] . $file['savename'];

        if (!$this->uploadReplace && is_file($filename)) {
            // 不覆盖同名文件
            $this->error = '文件已经存在！' . $filename;
            return false;
        }
        // 如果是图像文件 检测文件格式
        if (in_array(strtolower($file['extension']), array('gif', 'jpg', 'jpeg', 'bmp', 'png', 'swf')) && false === getimagesize($file['tmp_name'])) {
            $this->error = '非法图像文件';
            return false;
        }

        if (!move_uploaded_file($file['tmp_name'], $this->auto_charset($filename, 'utf-8', 'gbk'))) {
            $this->error = '文件上传保存错误！';
            return false;
        }
        if ($this->thumb && in_array(strtolower($file['extension']), array('gif', 'jpg', 'jpeg', 'bmp', 'png'))) {
            $image = getimagesize($filename);
            if (false !== $image) {
                //是图像文件生成缩略图
                $thumbWidth = explode(',', $this->thumbMaxWidth);
                $thumbHeight = explode(',', $this->thumbMaxHeight);
                $thumbPrefix = explode(',', $this->thumbPrefix);
                $thumbSuffix = explode(',', $this->thumbSuffix);
                $thumbFile = explode(',', $this->thumbFile);
                //20120318 修改默认缩图保存到和原图一个目录下
                $thumbPath = $this->thumbPath ? $this->thumbPath : $file['savepath'] . $this->getSubName($file) . "/";
                // 生成图像缩略图
                import($this->imageClassPath);
                $realFilename = $this->autoSub ? basename($file['savename']) : $file['savename'];
                for ($i = 0, $len = count($thumbWidth); $i < $len; $i++) {
                    $thumbname = $thumbPath . $thumbPrefix[$i] . substr($realFilename, 0, strrpos($realFilename, '.')) . $thumbSuffix[$i] . '.' . $file['extension'];
                    Image::thumb($filename, $thumbname, '', $thumbWidth[$i], $thumbHeight[$i], true);
                }
                if ($this->thumbRemoveOrigin) {
                    // 生成缩略图之后删除原图
                    //如果缩图前缀没有设置，将不删除原图，因为这时原图和缩图是一样的名字！
                    if (!empty($this->thumbPrefix)) {
                        unlink($filename);
                    }
                }
            }
        }
        if ($this->zipImags) {
            // TODO 对图片压缩包在线解压
        }
        return true;
    }

    /**
     *  把一个文件上传到FTP附件服务器上
     * @param type $upfile 本地存放地址，需要上传的文件
     * @param type $file 远程存放地址
     * @return string 
     */
    public function FTPuplode($upfile, $file) {
        if ($this->Ftpstatus) {
            // 远程存放地址
            $remote = CONFIG_FTPUPPAT . str_replace(SITE_PATH . "/", "", $file);
            //FTP上传
            if ($this->Ftp->put($remote, $upfile) == false) {
                $this->error = '远程附件上传失败！' . $this->Ftp->get_error();
                $this->Ftp->close();
                return false;
            }
        }
        return true;
    }

    /**
      +----------------------------------------------------------
     * 上传所有文件
      +----------------------------------------------------------
     * @access public
      +----------------------------------------------------------
     * @param array $Callback  回调参数
     * @param string $savePath  上传文件保存路径
      +----------------------------------------------------------
     * @return string
      +----------------------------------------------------------
     * @throws ThinkExecption
      +----------------------------------------------------------
     */
    public function upload($Callback = false ,$savePath = '') {
        //如果不指定保存文件名，则由系统默认
        if (empty($savePath))
            $savePath = $this->savePath;
        // 检查上传目录
        if (!is_dir($savePath)) {
            // 检查目录是否编码后的
            if (is_dir(base64_decode($savePath))) {
                $savePath = base64_decode($savePath);
            } else {
                // 尝试创建目录
                if (!mkdir($savePath)) {
                    $this->error = '上传目录' . $savePath . '不存在';
                    return false;
                }
            }
        } else {
            if (!is_writeable($savePath)) {
                $this->error = '上传目录' . $savePath . '不可写';
                return false;
            }
        }
        $fileInfo = array();
        $isUpload = false;

        // 获取上传的文件信息
        // 对$_FILES数组信息处理
        $files = $this->dealFiles($_FILES);
        foreach ($files as $key => $file) {
            //过滤无效的上传
            if (!empty($file['name'])) {
                //登记上传文件的扩展信息
                $file['key'] = $key;
                $file['extension'] = $this->getExt($file['name']);
                $file['savepath'] = $savePath;
                $file['savename'] = $this->getSaveName($file);

                // 自动检查附件
                if ($this->autoCheck) {
                    if (!$this->check($file))
                        return false;
                }

                //保存上传文件
                if (!$this->save($file))
                    return false;
                if (function_exists($this->hashType)) {
                    $fun = $this->hashType;
                    $file['hash'] = $fun($this->auto_charset(str_replace(C("UPLOADFILEPATH"), "", $file['savepath'] . $file['savename']), 'utf-8', 'gbk'));
                }

                //上传成功后保存文件信息，供其他地方调用
                unset($file['tmp_name'], $file['error']);
                $fileInfo[] = $file;
                $isUpload = true;
            }
        }
        if ($isUpload) {
            $this->uploadFileInfo = $fileInfo;
            //回调
            if ($Callback) {
                call_user_func_array($Callback[0], array($this, $this->uploadFileInfo , $Callback[1]));
            }
            //FTP
            if ($this->Ftpstatus) {
                foreach ($this->uploadFileInfo as $file) {
                    //文件地址
                    $filename = $file['savepath'] . $file['savename'];
                    // 远程存放地址
                    $remote =  str_replace(SITE_PATH . "/", "", $filename);
                    if($this->FTPuplode($filename, $remote)){
                        //上传成功删除原图
                        unlink($filename);
                    }else{
                        return false;
                    }
                }
            }
            return true;
        } else {
            $this->error = '没有选择上传文件';
            return false;
        }
    }

    /**
      +----------------------------------------------------------
     * 上传单个上传字段中的文件 支持多附件
      +----------------------------------------------------------
     * @access public
      +----------------------------------------------------------
     * @param array $file  上传文件信息
     * @param string $savePath  上传文件保存路径
      +----------------------------------------------------------
     * @return string
      +----------------------------------------------------------
     * @throws ThinkExecption
      +----------------------------------------------------------
     */
    public function uploadOne($file, $savePath = '') {
        //如果不指定保存文件名，则由系统默认
        if (empty($savePath))
            $savePath = $this->savePath;
        // 检查上传目录
        if (!is_dir($savePath)) {
            // 尝试创建目录
            if (!mk_dir($savePath)) {
                $this->error = '上传目录' . $savePath . '不存在';
                return false;
            }
        } else {
            if (!is_writeable($savePath)) {
                $this->error = '上传目录' . $savePath . '不可写';
                return false;
            }
        }
        //过滤无效的上传
        if (!empty($file['name'])) {
            $fileArray = array();
            if (is_array($file['name'])) {
                $keys = array_keys($file);
                $count = count($file['name']);
                for ($i = 0; $i < $count; $i++) {
                    foreach ($keys as $key)
                        $fileArray[$i][$key] = $file[$key][$i];
                }
            } else {
                $fileArray[] = $file;
            }
            $info = array();
            foreach ($fileArray as $key => $file) {
                //登记上传文件的扩展信息
                $file['extension'] = $this->getExt($file['name']);
                $file['savepath'] = $savePath;
                $file['savename'] = $this->getSaveName($file);
                // 自动检查附件
                if ($this->autoCheck) {
                    if (!$this->check($file))
                        return false;
                }
                //保存上传文件
                if (!$this->save($file))
                    return false;
                if (function_exists($this->hashType)) {
                    $fun = $this->hashType;
                    $file['hash'] = $fun($this->auto_charset($file['savepath'] . $file['savename'], 'utf-8', 'gbk'));
                }
                unset($file['tmp_name'], $file['error']);
                $info[] = $file;
            }
            // 返回上传的文件信息
            return $info;
        } else {
            $this->error = '没有选择上传文件';
            return false;
        }
    }

    // 自动转换字符集 支持数组转换
    private function auto_charset($fContents, $from = 'gbk', $to = 'utf-8') {
        $from = strtoupper($from) == 'UTF8' ? 'utf-8' : $from;
        $to = strtoupper($to) == 'UTF8' ? 'utf-8' : $to;
        if (strtoupper($from) === strtoupper($to) || empty($fContents) || (is_scalar($fContents) && !is_string($fContents))) {
            //如果编码相同或者非字符串标量则不转换
            return $fContents;
        }
        if (is_string($fContents)) {
            if (function_exists('mb_convert_encoding')) {
                return mb_convert_encoding($fContents, $to, $from);
            } elseif (function_exists('iconv')) {
                return iconv($from, $to, $fContents);
            } else {
                return $fContents;
            }
        } elseif (is_array($fContents)) {
            foreach ($fContents as $key => $val) {
                $_key = $this->auto_charset($key, $from, $to);
                $fContents[$_key] = $this->auto_charset($val, $from, $to);
                if ($key != $_key)
                    unset($fContents[$key]);
            }
            return $fContents;
        }
        else {
            return $fContents;
        }
    }

    /**
      +----------------------------------------------------------
     * 转换上传文件数组变量为正确的方式
      +----------------------------------------------------------
     * @access private
      +----------------------------------------------------------
     * @param array $files  上传的文件变量
      +----------------------------------------------------------
     * @return array
      +----------------------------------------------------------
     */
    private function dealFiles($files) {
        $fileArray = array();
        $n = 0;
        foreach ($files as $file) {
            if (is_array($file['name'])) {
                $keys = array_keys($file);
                $count = count($file['name']);
                for ($i = 0; $i < $count; $i++) {
                    foreach ($keys as $key)
                        $fileArray[$n][$key] = $file[$key][$i];
                    $n++;
                }
            } else {
                $fileArray[$n] = $file;
                $n++;
            }
        }
        return $fileArray;
    }

    /**
      +----------------------------------------------------------
     * 获取错误代码信息
      +----------------------------------------------------------
     * @access public
      +----------------------------------------------------------
     * @param string $errorNo  错误号码
      +----------------------------------------------------------
     * @return void
      +----------------------------------------------------------
     * @throws ThinkExecption
      +----------------------------------------------------------
     */
    protected function error($errorNo) {
        switch ($errorNo) {
            case 1:
                $this->error = '上传的文件超过了 php.ini 中 upload_max_filesize 选项限制的值';
                break;
            case 2:
                $this->error = '上传文件的大小超过了 HTML 表单中 MAX_FILE_SIZE 选项指定的值';
                break;
            case 3:
                $this->error = '文件只有部分被上传';
                break;
            case 4:
                $this->error = '没有文件被上传';
                break;
            case 6:
                $this->error = '找不到临时文件夹';
                break;
            case 7:
                $this->error = '文件写入失败';
                break;
            default:
                $this->error = '未知上传错误！';
        }
        return;
    }

    /**
      +----------------------------------------------------------
     * 根据上传文件命名规则取得保存文件名
      +----------------------------------------------------------
     * @access private
      +----------------------------------------------------------
     * @param string $filename 数据
      +----------------------------------------------------------
     * @return string
      +----------------------------------------------------------
     */
    private function getSaveName($filename) {
        $rule = $this->saveRule;
        if (empty($rule)) {//没有定义命名规则，则保持文件名不变
            $saveName = $filename['name'];
        } else {
            if (function_exists($rule)) {
                //使用函数生成一个唯一文件标识号
                $saveName = $rule() . "." . $filename['extension'];
            } else {
                //使用给定的文件名作为标识号
                $saveName = $rule . "." . $filename['extension'];
            }
        }
        if ($this->autoSub) {
            // 使用子目录保存文件
            $filename['savename'] = $saveName;
            $saveName = $this->getSubName($filename) . '/' . $saveName;
        }
        return $saveName;
    }

    /**
      +----------------------------------------------------------
     * 获取子目录的名称
      +----------------------------------------------------------
     * @access private
      +----------------------------------------------------------
     * @param array $file  上传的文件信息
      +----------------------------------------------------------
     * @return string
      +----------------------------------------------------------
     */
    private function getSubName($file) {
        switch ($this->subType) {
            case 'date':
                $dir = date($this->dateFormat, time());
                break;
            case 'hash':
            default:
                $name = md5($file['savename']);
                $dir = '';
                for ($i = 0; $i < $this->hashLevel; $i++) {
                    $dir .= $name{$i} . '/';
                }
                break;
        }
        if (!is_dir($file['savepath'] . $dir)) {
            mk_dir($file['savepath'] . $dir);
        }
        return $dir;
    }

    /**
      +----------------------------------------------------------
     * 检查上传的文件
      +----------------------------------------------------------
     * @access private
      +----------------------------------------------------------
     * @param array $file 文件信息
      +----------------------------------------------------------
     * @return boolean
      +----------------------------------------------------------
     */
    private function check($file) {
        if ($file['error'] !== 0) {
            //文件上传失败
            //捕获错误代码
            $this->error($file['error']);
            return false;
        }
        //文件上传成功，进行自定义规则检查
        //检查文件大小
        if (!$this->checkSize($file['size'])) {
            $this->error = '上传文件大小不符！';
            return false;
        }

        //检查文件Mime类型
        if (!$this->checkType($file['type'])) {
            $this->error = '上传文件MIME类型不允许！';
            return false;
        }
        //检查文件类型
        if (!$this->checkExt($file['extension'])) {
            $this->error = '上传文件类型不允许';
            return false;
        }

        //检查是否合法上传
        if (!$this->checkUpload($file['tmp_name'])) {
            $this->error = '非法上传文件！';
            return false;
        }
        return true;
    }

    /**
      +----------------------------------------------------------
     * 检查上传的文件类型是否合法
      +----------------------------------------------------------
     * @access private
      +----------------------------------------------------------
     * @param string $type 数据
      +----------------------------------------------------------
     * @return boolean
      +----------------------------------------------------------
     */
    private function checkType($type) {
        if (!empty($this->allowTypes))
            return in_array(strtolower($type), $this->allowTypes);
        return true;
    }

    /**
      +----------------------------------------------------------
     * 检查上传的文件后缀是否合法
      +----------------------------------------------------------
     * @access private
      +----------------------------------------------------------
     * @param string $ext 后缀名
      +----------------------------------------------------------
     * @return boolean
      +----------------------------------------------------------
     */
    private function checkExt($ext) {
        if (!empty($this->allowExts)) {
            //强制危险后缀过滤
            if (in_array(strtolower($ext), array("php", "php4", "asp", "phtml", "php3", "exe", "dll", "cer", "asa", "shtml", "shtm", "aspx", "asax", "cgi", "fcgi", "pl"))) {
                return false;
            }
            return in_array(strtolower($ext), $this->allowExts, true);
        }
        return false;
    }

    /**
      +----------------------------------------------------------
     * 检查文件大小是否合法
      +----------------------------------------------------------
     * @access private
      +----------------------------------------------------------
     * @param integer $size 数据
      +----------------------------------------------------------
     * @return boolean
      +----------------------------------------------------------
     */
    private function checkSize($size) {
        return !($size > $this->maxSize) || (-1 == $this->maxSize);
    }

    /**
      +----------------------------------------------------------
     * 检查文件是否非法提交
      +----------------------------------------------------------
     * @access private
      +----------------------------------------------------------
     * @param string $filename 文件名
      +----------------------------------------------------------
     * @return boolean
      +----------------------------------------------------------
     */
    private function checkUpload($filename) {
        return is_uploaded_file($filename);
    }

    /**
      +----------------------------------------------------------
     * 取得上传文件的后缀
      +----------------------------------------------------------
     * @access private
      +----------------------------------------------------------
     * @param string $filename 文件名
      +----------------------------------------------------------
     * @return boolean
      +----------------------------------------------------------
     */
    private function getExt($filename) {
        $pathinfo = pathinfo($filename);
        return $pathinfo['extension'];
    }

    /**
      +----------------------------------------------------------
     * 取得上传文件的信息
      +----------------------------------------------------------
     * @access public
      +----------------------------------------------------------
     * @return array
      +----------------------------------------------------------
     */
    public function getUploadFileInfo() {
        return $this->uploadFileInfo;
    }

    /**
      +----------------------------------------------------------
     * 取得最后一次错误信息
      +----------------------------------------------------------
     * @access public
      +----------------------------------------------------------
     * @return string
      +----------------------------------------------------------
     */
    public function getErrorMsg() {
        return $this->error;
    }

}